D = 20;
datafile = 'movielens.mat';
tag = 'ML';

wtfile = sprintf('%s_PTF2_D%d.mat', tag, D);
load(datafile);

fprintf('Starting Probabilistic Tensor Factorization (PTF) D = %d ...\n', D)
PTF2(train_data, probe_data, D, 50, tag);

load(wtfile);
fprintf('Starting Bayesian Probabilistic Tensor Factorization (BPTF) D = %d ...\n', D)
BPTF2(train_data,probe_data,D,50,PTF2_Wt,tag);
